import torch

# python -m ipykernel install --user --name=pytorch-gpu
# pip3 install torch torchvision torchaudio
# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
device = torch.device("cpu")

# logger.info(f"使用的计算设备为-->{device.type}")
# 执行对比算法时的线程池大小
thread_pool_size = 6
